/****************************************************************************
*
*    Copyright (c) 2017 - 2022 by Rockchip Corp.  All rights reserved.
*
*    The material in this file is confidential and contains trade secrets
*    of Rockchip Corporation. This is proprietary information owned by
*    Rockchip Corporation. No part of this work may be disclosed,
*    reproduced, copied, transmitted, or used in any way for any purpose,
*    without the express written permission of Rockchip Corporation.
*
*****************************************************************************/


#ifndef _YOLO_INFERENCE_API_H
#define _YOLO_INFERENCE_API_H

#include "rknn_api.h"
#include "yolo.h"
#include "resize_function.h"
#include "rknn_demo_utils.h"

#ifdef __cplusplus
extern "C" {
#endif

void yolo_init(char *model_path, char *yolo_type, const char *label_txt_path, int class_num, float conf_thresh, float nms_thresh);
detect_result_group_t yolo_infer_frame(unsigned char *inputData, int input_height, int input_width);
void push_detect_result_group(detect_result_group_t detect_result_group);
detect_result_group_t pop_detect_result_group();
void yolo_deinit();

#ifdef __cplusplus
} //extern "C"
#endif

#endif  //_RKNN_API_H
